import os
import shutil
import platform

def del_file(path_data):
    for i in os.listdir(path_data) :# os.listdir(path_data)#返回一个列表，里面是当前目录下面的所有东西的相对路径
        file_data = path_data + "/" + i#当前文件夹的下面的所有东西的绝对路径
        if os.path.isfile(file_data) == True:#os.path.isfile判断是否为文件,如果是文件,就删除.如果是文件夹.递归给del_file.
            os.remove(file_data)
        else:
            del_file(file_data)
def main():
    path = './tools/Download'
    path=path.strip()
    path=path.rstrip("\\")
    # 判断路径是否存在
    isExists=os.path.exists(path)
    # 不存在
    if isExists:
        # 如果存在目录，则删除存在的文件夹包括子文件
        del_file(path)
        os.rmdir(path+'/bootloader')
        os.rmdir(path+'/fastdown')
        os.rmdir(path+'/image')
        os.rmdir(path+'/progress')
        os.rmdir(path)
    #读取Python版本，根据版本选择相应的py下载文件
    py_ver = platform.python_version()
    if( py_ver[:3] == '3.9'):
        print('当前版本的 python =' + py_ver[:3])
        num = 39
        shutil.copytree('./tools/py39/', './tools/Download/')
    elif ( py_ver[:3] == '3.8'):
        print('当前版本的 python =' + py_ver[:3])
        num = 38
        shutil.copytree('./tools/py38/', './tools/Download/')
    else :
        print("你安装的python版本不对，请安装 python 3.8 或者 Python 3.9 版本")
        return 
    print('您好，python 下载环境已经配置完毕，现在可以使用python zos.py 进行相关操作了。')
   

# 调用函数
main()